﻿// TortoiseGit - a Windows shell extension for easy version control

// Copyright (C) 2016, 2023 - TortoiseGit
// Copyright (C) 2003-2007, 2010, 2013-2015 - TortoiseSVN

// This program is free software; you can redistribute it and/or
// modify it under the terms of the GNU General Public License
// as published by the Free Software Foundation; either version 2
// of the License, or (at your option) any later version.

// This program is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
// GNU General Public License for more details.

// You should have received a copy of the GNU General Public License
// along with this program; if not, write to the Free Software Foundation,
// 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
//

#pragma once
#include "ResString.h"

#include <string>


/**
 * \ingroup Utils
 * A base window class.
 * Provides separate window message handlers for every window object based on
 * this class.
 */
class CWindow
{
public:
	virtual bool RegisterWindow(UINT style, HICON hIcon, HCURSOR hCursor,
								HBRUSH hbrBackground, LPCWSTR lpszMenuName,
								LPCWSTR lpszClassName, HICON hIconSm);
	virtual bool RegisterWindow(CONST WNDCLASSEX* wcx);

	/// static message handler to put in WNDCLASSEX structure
	static LRESULT CALLBACK stWinMsgHandler(HWND hwnd, UINT uMsg, WPARAM wParam, LPARAM lParam);

	/**
	 * Sets the window title.
	 */
	void SetWindowTitle(const std::wstring& sTitle)
	{
		sWindowTitle = sTitle;
		if (m_hwnd)
			SetWindowText(m_hwnd, sWindowTitle.c_str());
	};

	void SetRegistryPath(const std::wstring& sPath)
	{
		size_t slashPos = sPath.find_last_of('\\');
		sRegistryPath = sPath.substr(0, slashPos);
		sRegistryValue = sPath.substr(slashPos+1);
	}

	/**
	 * Sets the transparency of the window.
	 * \remark note that this also sets the WS_EX_LAYERED style!
	 */
	void SetTransparency(BYTE alpha, COLORREF color = 0xFF000000);

	virtual bool Create();
	virtual bool Create(DWORD dwStyles, HWND hParent = nullptr, RECT* rect = nullptr);
	virtual bool CreateEx(DWORD dwExStyles, DWORD dwStyles, HWND hParent = nullptr, RECT* rect = nullptr, LPCWSTR classname = nullptr);

	//void MsgLoop();
	bool IsWindowClosed() const { return bWindowClosed; };

	operator HWND() const {return m_hwnd;}
protected:
	HINSTANCE hResource = nullptr;
	HWND m_hwnd = nullptr;
	HWND m_hParent = nullptr;
	bool bWindowClosed = false;
	std::wstring sClassName;
	std::wstring sWindowTitle;
	std::wstring sRegistryPath;
	std::wstring sRegistryValue;
	bool bWindowRestored = false;

	//constructor
	CWindow(HINSTANCE hInstance, CONST WNDCLASSEX* wcx = nullptr)
		: hResource(hInstance)
	{
		if (wcx)
			RegisterWindow(wcx);
	};

	// the real message handler
	virtual LRESULT CALLBACK WinMsgHandler(HWND hwnd, UINT uMsg, WPARAM wParam, LPARAM lParam) = 0;

	// returns a pointer the window (stored as the WindowLong)
	inline static CWindow * GetObjectFromWindow(HWND hWnd)
	{
		return reinterpret_cast<CWindow*>(GetWindowLongPtr(hWnd, GWLP_USERDATA));
	}
};
